from trainv7 import *

if __name__ == "__main__":
    torch.cuda.empty_cache()
    opt = parse_opt()
    # opt.weights = './yolo/weights/yolov7-tiny.pt'
    opt.data = './cjc/data/head.yaml'
    opt.batch_size=1
    opt.resume = False
    opt.name = 'head'
    try:
        main(opt)
    except KeyboardInterrupt as e:
        print("手动退出")


